tensor([[[[[0.0000, 0.5900],
           [0.5900, 0.6200]],

          [[0.0000, 0.3100],
           [0.3100, 0.0000]]],


         [[[0.0000, 0.0100],
           [0.0000, 1.0000]],

          [[0.3100, 0.3100],
           [0.3100, 0.3100]]]],



        [[[[0.6200, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.2800],
           [0.2800, 0.5000]]],


         [[[0.5500, 0.5300],
           [0.5000, 0.0000]],

          [[0.2800, 0.2800],
           [0.2800, 0.2800]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.2200],
           [0.2200, 0.5000]]],


         [[[0.0000, 0.0000],
           [0.5000, 0.0000]],

          [[0.2100, 0.2000],
           [0.2000, 0.2100]]]],



        [[[[0.3800, 0.4100],
           [0.4100, 0.3800]],

          [[1.0000, 0.1800],
           [0.1800, 0.0000]]],


         [[[0.4500, 0.4600],
           [0.0000, 0.0000]],

          [[0.2000, 0.2000],
           [0.2000, 0.2000]]]]], grad_fn=<RoundBackward1>)